# 学员信息在线管理
import DAO
# 定义一个用于存放学员信息的列表变量

#定义一个学生信息的输出函数
def showStu(stulist):
    '''
    学生信息的输出函数
    '''
    if len(stulist)==0:
        print("========== 没有学员信息可以输出！=============")
        return
    print("|{0:<5}| {1:<10}| {2:<5}| {3:<10}|".format("sid","name","age","classid"))
    print("-"*40)
    for stuinfo in stulist:
        print("|{0:<5}| {1:<10}| {2:<5}| {3:<10}|".format(stuinfo[0],stuinfo[1],stuinfo[2],stuinfo[3]))


while True:
    dao = DAO.dao('studb')
    # 输出初始界面
    print("="*12,"学员管理系统","="*14)
    print("{0:1} {1:13} {2:15}".format(" ","1. 查看学员信息","2. 添加学员信息"))
    print("{0:1} {1:13} {2:15}".format(" ","3. 删除学员信息","4. 退出系统"))
    print("="*40)
    key = input("请输入对应的选择：")
    # 根据键盘值，判断并执行对应的操作
    if key == "1":
        print("="*12,"学员信息浏览","="*14)
        showStu(dao.queryall())
        input("按回车键继续：")
    elif key == "2":
        print("="*12,"学员信息添加","="*14)
        name=input("请输入要添加的姓名：")
        age=input("请输入要添加的年龄：")
        classid=input("请输入要添加的班级号：")
        dao.insertStu((name,int(age),classid))
        showStu(dao.queryall())
        input("按回车键继续：")
    elif key == "3":
        print("="*12,"学员信息删除","="*14)
        showStu(dao.queryall())
        sid = input("请输入你要删除的信息id号：")
        dao.delStu(int(sid))
        showStu(dao.queryall())
        input("按回车键继续：")
    elif key == "4":
        print("="*12,"再见","="*14)
        break
    else:
        print("======== 无效的键盘输入！ ==========")